Put the ipynb file and html file in the github branch you created in the last assignment and submit the link to the commit in brightspace
from plotly.offline import init_notebook_mode
import plotly.io as pio
import plotly.express as px
import pandas as pd
import numpy as np
init_notebook_mode(connected=True)
pio.renderers.default = "plotly_mimetype+notebook"
#load data
df = px.data.gapminder()
df.head()
| country | continent | year | lifeExp | pop | gdpPercap | iso_alpha | iso_num | |
|---|---|---|---|---|---|---|---|---|
| 0 | Afghanistan | Asia | 1952 | 28.801 | 8425333 | 779.445314 | AFG | 4 |
| 1 | Afghanistan | Asia | 1957 | 30.332 | 9240934 | 820.853030 | AFG | 4 |
| 2 | Afghanistan | Asia | 1962 | 31.997 | 10267083 | 853.100710 | AFG | 4 |
| 3 | Afghanistan | Asia | 1967 | 34.020 | 11537966 | 836.197138 | AFG | 4 |
| 4 | Afghanistan | Asia | 1972 | 36.088 | 13079460 | 739.981106 | AFG | 4 |
Recreate the barplot below that shows the population of different continents for the year 2007.
Hints:
df1 = df[df.year == 2007] # Extract 2007 data
df2 = pd.DataFrame(df1.groupby(by=df.continent).sum()) # Sum the population per continent
dfnew = pd.DataFrame(df2['pop'])
fig = px.bar(dfnew, # Use plotly bar
title='Population per continent (2007)',
x='pop',
y=dfnew.index,
orientation='h',
color=dfnew.index) # Add different colors for different continents
fig.show()
fig.update_yaxes(categoryorder='total ascending') # Sort order of continents
fig.show()
Add text to each bar that represents the population
fig = px.bar(dfnew,
title='Population per continent (2007)',
x='pop',
y=dfnew.index,
orientation='h',
color=dfnew.index,
text_auto='.2s') # Add text to each bar that represents the population
fig.update_traces(textposition='outside')
fig.show()
Thus far we looked at data from one year (2007). Lets create an animation to see the population growth of the continents through the years
fig = px.bar(df, x="pop", y="continent", color="continent",
animation_frame="year", animation_group="country", range_x=[0,4*10**9])
fig.update_layout(barmode='stack')
fig.update_traces(marker_line_width=0)
fig.update_yaxes(categoryorder='total ascending')
fig.show()
Instead of the continents, lets look at individual countries. Create an animation that shows the population growth of the countries through the years
fig = px.bar(df, x="pop", y="country", color="country",
animation_frame="year", animation_group="country",
range_x=[0, 1.5*10**9])
fig.update_layout(barmode='stack')
fig.update_traces(marker_line_width=0)
fig.update_yaxes(categoryorder='total ascending')
fig.show()
Clean up the country animation. Set the height size of the figure to 1000 to have a better view of the animation
fig.update_layout(height=1000)
fig.show()
fig.update_yaxes(range=(len(df.country.unique())-10.5,
len(df.country.unique())-0.5))
fig.update_layout(height=600)
fig.show()